import time

from selenium import webdriver


def startUp():
    # 驱动浏览器（第一个字母必须大写）
    driver = webdriver.Chrome()
    # 访问目标网址（必须是以http开头的全网址）
    driver.get("http://www.baidu.com")
    # 最大化浏览器页面
    driver.maximize_window()
    time.sleep(3)
    # 后退（访问上一步访问的网址）
    driver.back()
    time.sleep(3)
    # 前进（访问后一步访问的网址）
    driver.forward()
    time.sleep(3)
    # 刷新浏览器
    driver.refresh()
    time.sleep(3)
    # 关闭当前页面
    driver.close()
    # 截图
    driver.get_screenshot_as_file("./baidu.png")
    png = driver.get_screenshot_as_png()
    with open("./baidu2.png", "wb") as fp:
        fp.write(png)
        time.sleep(3)
        # 退出浏览器
        driver.quit()


if __name__ == '__main__':
    startUp()
